package com.tanhua.autoconfig.template;

import com.baidu.aip.face.AipFace;
import org.json.JSONObject;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.HashMap;

/**
 * 人脸识别模板
 */

public class AipTemplate {
    @Autowired
    private AipFace client;

    /**
     * 检测图片是否包含人脸，如果包含返回true，不包含返回false
     * 传入图片路径
     * @param url
     * @return
     */
    public boolean recognition(String url) {
        // 传入可选参数调用接口
        HashMap<String, String> options = new HashMap<String, String>();
        options.put("face_field", "age");   //默认只返回face_token、人脸框、概率和旋转角度
        options.put("max_face_num", "2");   //最多处理人脸的数目
        options.put("face_type", "LIVE");   //人脸的类型 LIVE表示生活照
        options.put("liveness_control", "LOW"); //活体检测控制
        //图片路径和类型
        String imageType = "URL";
        // 人脸检测
        JSONObject res = client.detect(url, imageType, options);
        System.out.println(res.toString(2));

        //如果错误码为0，则说明人脸验证成功
        Integer error_code = (Integer) res.get("error_code");
        return error_code == 0;

    }

}
